/*******************************************************************************
 * Copyright 2009 DCSpectrometer - http://code.google.com/p/dcspectrometer 
 *  
 * Licensed under the Apache License, Version 2.0 (the "License"); 
 * you may not use this file except in compliance with the License. 
 * You may obtain a copy of the License at 
 *  
 *     http://www.apache.org/licenses/LICENSE-2.0 
 *     
 * Unless required by applicable law or agreed to in writing, software 
 * distributed under the License is distributed on an "AS IS" BASIS, 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
 * See the License for the specific language governing permissions and 
 * limitations under the License. 
 *******************************************************************************/
package com.dcspectrometer.kmeansclustering;

import java.awt.Point;
import java.awt.geom.Point2D;

@SuppressWarnings("serial")
public class ClusterCentroid extends Point2D.Double {
  private Cluster _cluster;

  public ClusterCentroid(double x, double y) {
    super(x, y);
    _cluster = null;
  }

  public ClusterCentroid(Point point) {
    super(point.x, point.y);
    _cluster = null;
  }

  public void recalculateCentroid() {
    /** Find new location of cluster centroid. */
    this.x = 0;
    this.y = 0;
    for (int i = 0; i < _cluster.getPointCount(); i++) {
      Point nextPoint = _cluster.getPoint(i).getLocation();
      this.x += nextPoint.x;
      this.y += nextPoint.y;
    }
    this.x /= _cluster.getPointCount();
    this.y /= _cluster.getPointCount();

    /** Recalculate cluster energy. */
    _cluster.calcClusterEnergy();
  }

  public void setCluster(Cluster cluster) {
    this._cluster = cluster;
  }

  public Cluster getCluster() {
    return _cluster;
  }
}
